
"""
@author: xiangping
@contact: xiangpingbu@gmail.com
@time: 2020/3/4 4:15 下午
@file: PickleModel
@Desc:
"""
import pickle

from app.configs.model_path import MODEL_DICT
from app.services.models.base_model import BaseModel


class PickleModel(BaseModel):
    def load_model(self):
        return pickle.load(open(MODEL_DICT[self.model_name], "rb"))